package 每日一题;

import com.alibaba.fastjson.JSON;
import javafx.util.Pair;

import java.util.Arrays;
import java.util.Stack;

public class No503下一个更大元素II {

    /**
     * 给定一个循环数组（最后一个元素的下一个元素是数组的第一个元素），
     * 输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，
     * 这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。
     *
     * 示例 1:
     * 输入: [1,2,1]
     * 输出: [2,-1,2]
     * 解释: 第一个 1 的下一个更大的数是 2；
     * 数字 2 找不到下一个更大的数；
     * 第二个 1 的下一个最大的数需要循环搜索，结果也是 2。
     * 注意: 输入数组的长度不会超过 10000。
     */

    public int[] nextGreaterElements(int[] nums) {

        if(nums.length<1){
            return new int[]{};
        }

        int maxIndex=0;
        int maxNum=nums[0];

        for (int i = 1; i < nums.length; i++) {
            if(nums[i]>maxNum){
                maxNum=nums[i];
                maxIndex=i;
            }
        }

        //从maxIndex后一位开始单调栈
        Stack<Integer> stack=new Stack<>();
        int[] result=new int[nums.length];
        Arrays.fill(result,-1);

        //优化点:可以不使用Pair,只存入下标,peek时将下标带入数组即可
        for (int i = maxIndex+1; i < nums.length; i++) {
            if(stack.isEmpty()||nums[stack.peek()]>=nums[i]){
                stack.add(i);
            }else{
                while (!stack.isEmpty()&&nums[stack.peek()]<nums[i]){
                    Integer index = stack.pop();
                    result[index]=nums[i];
                }
                stack.add(i);
            }
        }

        for (int i = 0; i <= maxIndex; i++) {
            if(stack.isEmpty()||nums[stack.peek()]>=nums[i]){
                stack.add(i);
            }else{
                while (!stack.isEmpty()&&nums[stack.peek()]<nums[i]){
                    Integer index = stack.pop();
                    result[index]=nums[i];
                }
                stack.add(i);
            }
        }

        return result;
    }

}
